"""
首页界面
"""
from selenium.webdriver.common.by import By

from base.base_page import BasePage


class IndexPage(BasePage):
    """
    首页类
    """

    def __init__(self):
        super().__init__(driver_type='web')
        # 登录按钮元素定位
        self.login_btn = By.CLASS_NAME, 'red'
        # 搜索框元素定位
        self.search_box = By.ID, 'q'
        # 搜索按键元素定位
        self.goods_btn = By.CLASS_NAME, 'ecsc-search-button'
        # 购物车元素定位
        self.mg_cart_btn = By.CLASS_NAME, 'c-n'

    def go_to_login_page(self):
        """
        跳转登录页面
        """
        self.click_element(self.find_element_method(self.login_btn))

    def search_goods_page(self, search_name):
        """
        搜索商品方法
        :param search_name: 商品名字
        """
        self.input_element(self.find_element_method(self.search_box), search_name)
        self.click_element(self.find_element_method(self.goods_btn))

    def go_to_mg_cart_page(self):
        """
        跳转我的购物车
        """
        self.click_element(self.find_element_method(self.mg_cart_btn))


if __name__ == '__main__':
    IndexPage().go_to_login_page()
